package com.ruoyi.generator.service.impl;

import java.sql.SQLException;
import java.util.List;

import com.ruoyi.common.exception.ServiceException;
import com.ruoyi.common.utils.DateUtils;
import com.ruoyi.common.utils.aes.AESEncryptor;
import com.ruoyi.generator.sdk.MySqlConnectionClient;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;
import com.ruoyi.generator.mapper.DatabaseMapper;
import com.ruoyi.generator.domain.Database;
import com.ruoyi.generator.service.IDatabaseService;

/**
 * 数据源信息Service业务层处理
 * 
 * @author ruoyi
 * @date 2024-12-26
 */
@Service
public class DatabaseServiceImpl implements IDatabaseService 
{
    @Autowired
    private DatabaseMapper databaseMapper;

    @Autowired
    private AESEncryptor aesEncryptor;
    /**
     * 查询数据源信息
     * 
     * @param id 数据源信息主键
     * @return 数据源信息
     */
    @Override
    public Database selectDatabaseById(Long id)
    {
        return databaseMapper.selectDatabaseById(id);
    }

    /**
     * 查询数据源信息列表
     * 
     * @param database 数据源信息
     * @return 数据源信息
     */
    @Override
    public List<Database> selectDatabaseList(Database database)
    {
        return databaseMapper.selectDatabaseList(database);
    }

    /**
     * 新增数据源信息
     * 
     * @param database 数据源信息
     * @return 结果
     */
    @Override
    public int insertDatabase(Database database)
    {
        database.setCreateTime(DateUtils.getNowDate());
        return databaseMapper.insertDatabase(database);
    }

    /**
     * 修改数据源信息
     * 
     * @param database 数据源信息
     * @return 结果
     */
    @Override
    public int updateDatabase(Database database)
    {
        database.setUpdateTime(DateUtils.getNowDate());
        return databaseMapper.updateDatabase(database);
    }

    /**
     * 批量删除数据源信息
     * 
     * @param ids 需要删除的数据源信息主键
     * @return 结果
     */
    @Override
    public int deleteDatabaseByIds(Long[] ids)
    {
        return databaseMapper.deleteDatabaseByIds(ids);
    }

    /**
     * 删除数据源信息信息
     * 
     * @param id 数据源信息主键
     * @return 结果
     */
    @Override
    public int deleteDatabaseById(Long id)
    {
        return databaseMapper.deleteDatabaseById(id);
    }

    @Override
    public String testDatabase(Database database) {
        String password = aesEncryptor.decrypt(database.getBasePassword());
        MySqlConnectionClient mySqlConnectionClient = new MySqlConnectionClient.Builder()
                .mySqlIp(database.getBaseIp())//Ip
                .mySqlPort(database.getBasePort())//端口
                .databaseName(database.getBaseName())//数据库名
                .databasePassword(password)//数据库密码
                .databaseUsername(database.getBaseUsername())//数据库用户名
                .build();
        boolean b = false;
        try {
            b = mySqlConnectionClient.testConnection();
        } catch (SQLException e) {
            return "链接数据库"+database.getBaseName()+"失败！";
        }
        if (b){
            return "链接数据库"+database.getBaseName()+"成功！";
        }

        return "链接数据库"+database.getBaseName()+"失败！";
    }
}
